/*
你有一个单词列表 words 和一个模式  pattern，你想知道 words 中的哪些单词与模式匹配。

如果存在字母的排列 p ，使得将模式中的每个字母 x 替换为 p(x) 之后，我们就得到了所需的单词，那么单词与模式是匹配的。

（回想一下，字母的排列是从字母到字母的双射：每个字母映射到另一个字母，没有两个字母映射到同一个字母。）

返回 words 中与给定模式匹配的单词列表。

你可以按任何顺序返回答案。

 

示例：

输入：words = ["abc","deq","mee","aqq","dkd","ccc"], pattern = "abb"
输出：["mee","aqq"]
解释：
"mee" 与模式匹配，因为存在排列 {a -> m, b -> e, ...}。
"ccc" 与模式不匹配，因为 {a -> c, b -> c, ...} 不是排列。
因为 a 和 b 映射到同一个字母。
 

提示：

1 <= words.length <= 50
1 <= pattern.length = words[i].length <= 20

来源：力扣（LeetCode）
链接：https://leetcode.cn/problems/find-and-replace-pattern
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。
*/

#include "../stdc++.h"

// 双射
class Solution {
public:
    vector<string> findAndReplacePattern(vector<string>& words, string pattern) {
        vector<string> res;
        for (const string& word : words) {
            if (isModePattern(word, pattern)) {
                res.push_back(word);
            }
        }
        return res;
    }
private:
    bool isModePattern(const string& word, const string& pattern) {
        int m = word.size();
        int n = pattern.size();
        if (m != n) {
            return false;
        }
        unordered_map<char, char> pattern2Word;
        unordered_map<char, char> word2Pattern;
        for (int i = 0; i < m; ++i) {
            if (!pattern2Word.count(pattern[i]) && !word2Pattern.count(word[i])) {
                pattern2Word[pattern[i]] = word[i];
                word2Pattern[word[i]] = pattern[i];
            } else if (pattern2Word[pattern[i]] != word[i] || word2Pattern[word[i]] != pattern[i]) {
                return false;
            }
        }
        return true;
    }
};
